Python で Redshift の JDBC ドライバを使ってみる（JayDeBeApi）
Python から Redshift へ接続する方法で psycopg 以外の方法が無いものか探していた所 JayDeBeApi なるものを見つけました。Amazon 謹製の JDBC Driver を使えるかなということで試してみました。
テスト環境
- OS X 10.11.6, Java 8u112, Python 3.6.1, JayDeBeApi 1.1.1, RedshiftJDBC42-1.2.1.1001.jar
- Windows 7, Java 8u131, Python 2.7.13, JayDeBeApi 1.1.1, RedshiftJDBC42-1.2.1.1001.jar
※環境変数 JAVA_HOME の設定が必要です。
※本来 Python 3 の場合は JayDeBeApi3 を利用するようなのですが、エラーが発生してしまったので 1.1.1 で試しました。
Redshift JDBC ドライバの用意
コチラからダウンロードできます。任意の場所へ配置します。
JayDeBeApi のインストール
コチラの通り pip を用いてインストールできます。
$ pip install JayDeBeApi
Python インタプリタから Redshift へ接続
下記のように Redshift へ接続し一時テーブルを作成しました。
$ python
Python 3.6.1 (default, Jun 2 2017, 19:53:52)
[GCC 4.2.1 Compatible Apple LLVM 8.0.0 (clang-800.0.42.1)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import jaydebeapi
>>> conn = jaydebeapi.connect("com.amazon.redshift.jdbc.Driver",
... "jdbc:redshift://mycluster.us-east-1.redshift.amazonaws.com:5439/mydb",
... ["user", "password"],
... "/path/to/RedshiftJDBC42-1.2.1.1001.jar")
>>> curs = conn.cursor()
>>> curs.execute("create temp table item (name varchar(255), price integer, description text)")
>>> curs.execute("insert into item values ('hoge', 1234, 'HogeHoge')")
>>> curs.execute("insert into item values ('piyo', 2345, 'PiyoPiyo')")
>>> curs.execute("insert into item values ('fuga', 3456, 'FugaFuga')")
>>> curs.execute("select * from item")
SELECT 文を execute した後、fetchall で結果が取得できました。
>>> curs.execute("select * from item")
>>> curs.fetchall()
[('hoge', 1234, 'HogeHoge'), ('piyo', 2345, 'PiyoPiyo'), ('fuga', 3456, 'FugaFuga')]
>>> curs.close()
>>> conn.close()
>>> exit()
$
Windows の場合は Redshift JDBC ドライバのパス指定時にバックスラッシュのエスケープに気をつけてください。
# C:\path\to\RedshiftJDBC42-1.2.1.1001.jar "C:\\Path\\to\\RedshiftJDBC42-1.2.1.1001.jar"
まとめ
おもいのほか問題なく接続できたのでびっくりしました。Java 自体がマルチプラットフォームなので環境を選ばずに使用できるのがよいですね。Python から Redshift へクエリを発行したいケースで、選択肢のひとつになるではないかと思います。
